package com.crm.util;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.bson.types.ObjectId;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;

public class MongoDBUtil {
	  
	  
    private static final String HOST = "60.205.136.122";  
  
    private static final String dbName = "zhlog";  
  
    private static Mongo mongo;  
  
    private static DB db;  
  
    static {  
        try {  
            mongo = new Mongo(HOST);  
            db = mongo.getDB(dbName);  
            // db.authenticate(username, passwd)  
        } catch (MongoException e) {  
            e.printStackTrace();  
        }  
    }  
  
    private MongoDBUtil() {  
    }  
  
    /**
     * 
     * @Description : 添加
     * @author zealot
     * @param @param map
     * @param @param collectionName
     * @return void
     * @version 创建时间 : 2016年12月1日 上午10:06:42
     * @thorws
     */
    public static void add(Map<String, Object> map, String collectionName) {  
        DBObject dbObject = new BasicDBObject(map);  
        getCollection(collectionName).insert(dbObject);  
    }  
      
    /**
     * 
     * @Description : 添加
     * @author zealot
     * @param @param list
     * @param @param collectionName
     * @return void
     * @version 创建时间 : 2016年12月1日 上午10:06:55
     * @thorws
     */
    public static void add(List<Map<String, Object>> list, String collectionName) {  
        for (Map<String, Object> map : list) {  
            add(map, collectionName);  
        }  
    }  
      
    /**
     * 
     * @Description : 删除
     * @author zealot
     * @param @param map
     * @param @param collectionName
     * @return void
     * @version 创建时间 : 2016年12月1日 上午10:07:05
     * @thorws
     */
    public static void delete(Map<String, Object> map, String collectionName) {  
        DBObject dbObject = new BasicDBObject(map);  
        getCollection(collectionName).remove(dbObject);  
    }  
      
    /**
     * 
     * @Description : 根据key删除
     * @author zealot
     * @param @param id
     * @param @param collectionName
     * @return void
     * @version 创建时间 : 2016年12月1日 上午10:07:18
     * @thorws
     */
    public static void delete(String id, String collectionName) {  
        Map<String, Object> map = new HashMap<String, Object>();  
        map.put("_id", new ObjectId(id));  
        delete(map, collectionName);  
    }  
      
    /**
     * 
     * @Description : 删除全部
     * @author zealot
     * @param @param collectionName
     * @return void
     * @version 创建时间 : 2016年12月1日 上午10:07:30
     * @thorws
     */
    public static void deleteAll(String collectionName) {  
        getCollection(collectionName).drop();  
    }  
      
    /** 
     * 修改操作</br> 
     * 会用一个新文档替换现有文档,文档key结构会发生改变</br> 
     * 比如原文档{"_id":"123","name":"zhangsan","age":12}当根据_id修改age  
     * value为{"age":12}新建的文档name值会没有,结构发生了改变 
     * <br>------------------------------<br> 
     * @param whereMap       
     * @param valueMap       
     * @param collectionName 
     */  
    public static void update(Map<String, Object> whereMap, Map<String, Object> valueMap, String collectionName) {  
        executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){  
            public DBObject doCallback(DBObject valueDBObject) {  
                return valueDBObject;  
            }  
        });  
    }   
      
    /** 
     * 修改操作,使用$set修改器</br> 
     * 用来指定一个键值,如果键不存在,则自动创建,会更新原来文档, 不会生成新的, 结构不会发生改变 
     * <br>------------------------------<br> 
     * @param whereMap       
     * @param valueMap       
     * @param collectionName 
     */  
    public static void updateSet(Map<String, Object> whereMap, Map<String, Object> valueMap, String collectionName) {  
        executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){  
            public DBObject doCallback(DBObject valueDBObject) {  
                return new BasicDBObject("$set", valueDBObject);  
            }  
        });  
    }   
      
    /** 
     * 修改操作,使用$inc修改器</br> 
     * 修改器键的值必须为数字</br> 
     * 如果键存在增加或减少键的值, 如果不存在创建键 
     * <br>------------------------------<br> 
     * @param whereMap       
     * @param valueMap       
     * @param collectionName 
     */  
    public static void updateInc(Map<String, Object> whereMap, Map<String, Integer> valueMap, String collectionName) {  
        executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){  
            public DBObject doCallback(DBObject valueDBObject) {  
                return new BasicDBObject("$inc", valueDBObject);  
            }  
        });  
    }   
      
    /** 
     * 修改 
     * <br>------------------------------<br> 
     * @param collectionName 
     * @param whereMap 
     * @param valueMap 
     * @param updateCallback 
     */  
    private static void executeUpdate(String collectionName, Map whereMap, Map valueMap, UpdateCallback updateCallback) {  
        DBObject whereDBObject = new BasicDBObject(whereMap);  
        DBObject valueDBObject = new BasicDBObject(valueMap);  
        valueDBObject = updateCallback.doCallback(valueDBObject);  
        getCollection(collectionName).update(whereDBObject, valueDBObject);  
    }  
      
    interface UpdateCallback {  
          
        DBObject doCallback(DBObject valueDBObject);  
    }  
      
    /** 
     * 获取集合(表) 
     * <br>------------------------------<br> 
     * @param collectionName 
     * @return 
     */  
    public static DBCollection getCollection(String collectionName) {  
        return db.getCollection(collectionName);  
    }  
}
